package 中等.树;

import util.TreeNode;

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Deque;
import java.util.List;

/**
 * 给你 root1 和 root2 这两棵二叉搜索树。请你返回一个列表
 * ，其中包含 两棵树 中的所有整数并按 升序 排序。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/all-elements-in-two-binary-search-trees/
 */
public class 两棵二叉搜索树中的所有元素_1305 {

    public static void main(String[] args) {

    }

    /**
     * 队列+中序遍历
     * <p>
     * 使用list模拟队列要快一点点
     *
     * @param root1
     * @param root2
     * @return
     */
    public List<Integer> getAllElements(TreeNode root1, TreeNode root2) {
        List<Integer> list = new ArrayList<>();
        Deque<Integer> queue1 = new ArrayDeque<>();
        Deque<Integer> queue2 = new ArrayDeque<>();
        dfs(root1, queue1);
        dfs(root2, queue2);

        while (!queue1.isEmpty() || !queue2.isEmpty()) {
            if (queue1.isEmpty()) {
                list.add(queue2.pollFirst());
            } else if (queue2.isEmpty()) {
                list.add(queue1.pollFirst());
            } else {
                if (queue1.peekFirst() < queue2.peekFirst()) {
                    list.add(queue1.pollFirst());
                } else {
                    list.add(queue2.pollFirst());
                }
            }
        }
        return list;
    }

    private void dfs(TreeNode node, Deque<Integer> queue) {
        if (node == null) {
            return;
        }
        dfs(node.left, queue);
        queue.addLast(node.val);
        dfs(node.right, queue);
    }

}
